nanopyx.core.utils.timeit

View Source

 1import time
 2import types
 3
 4
 5def timeit(func: types.FunctionType):
 6
 7    def wrapper(*args, **kwargs):
 8        t = time.time()
 9        retval = func(*args, **kwargs)
10        print(f"{func.__name__} took {round(time.time()-t,3)} seconds")
11        return retval
12
13    return wrapper
14
15
16def timeit2(func):
17    def wrapper(*args, **kwargs):
18        start = time.time()
19        result = func(*args, **kwargs)
20        end = time.time()
21        delta = end - start
22        if delta < 10**-4:
23            msg = f"{func.__name__} took {delta/1e-6:.6f} nseconds"
24        elif delta < 10**-1:
25            msg = f"{func.__name__} took {delta/1e-3:.6f} mseconds"
26        else:
27            msg = f"{func.__name__} took {delta:.6f} seconds"
28        print(msg)
29        return result
30    return wrapper

def timeit(func: function): View Source

 6def timeit(func: types.FunctionType):
 7
 8    def wrapper(*args, **kwargs):
 9        t = time.time()
10        retval = func(*args, **kwargs)
11        print(f"{func.__name__} took {round(time.time()-t,3)} seconds")
12        return retval
13
14    return wrapper

def timeit2(func): View Source

17def timeit2(func):
18    def wrapper(*args, **kwargs):
19        start = time.time()
20        result = func(*args, **kwargs)
21        end = time.time()
22        delta = end - start
23        if delta < 10**-4:
24            msg = f"{func.__name__} took {delta/1e-6:.6f} nseconds"
25        elif delta < 10**-1:
26            msg = f"{func.__name__} took {delta/1e-3:.6f} mseconds"
27        else:
28            msg = f"{func.__name__} took {delta:.6f} seconds"
29        print(msg)
30        return result
31    return wrapper